package com.leetcode;

/**
 * n 皇后问题 研究的是如何将 n 个皇后放置在 n × n 的棋盘上，并且使皇后彼此之间不能相互攻击。
 * <p>
 * 给你一个整数 n ，返回 n 皇后问题 不同的解决方案的数量。
 */
public class No52 {
    public static int totalNQueens(int n) {
        int[] result = new int[1];
        int[] chess = new int[n];
        solve(0, chess, result);
        return result[0];
    }

    public static void solve(int index, int[] chess, int[] result) {
        if (index == chess.length) {
            result[0]++;
            return;
        }
        for (int i = 0; i < chess.length; i++) {
            chess[index] = i;
            if (judge(chess, index)) {
                solve(index + 1, chess, result);
            }
        }
    }

    public static boolean judge(int[] chess, int n) {
        for (int i = 0; i < n; i++) {
            if (chess[i] == chess[n] || Math.abs(chess[i] - chess[n]) == Math.abs(n - i)) {
                return false;
            }
        }
        return true;
    }

    public static void main(String[] args) {
        System.out.println(totalNQueens(4));
    }
}
